const express = require('express');
const cors = require('cors');
const axios = require('axios');

const app = express();
app.use(cors());
app.use(express.json()); // 支持 JSON 请求体

// 示例：统一处理前端传来的"调用AI"请求
app.post('/api/call-ai', async (req, res) => {
  const { modelName, prompt } = req.body;
  try {
    const result = await callAIModel(modelName, prompt);
    res.json({ success: true, data: result });
  } catch (error) {
    res.json({ success: false, error: error.message });
  }
});

// 统一管理AI API调用
async function callAIModel(modelName, prompt) {
  // 后面这里根据modelName不同，调用不同的AI平台API
  if (modelName === '通义千问') {
    const response = await axios.post('通义API地址', {
      prompt: prompt
    }, {
      headers: {
        Authorization: 'Bearer 你的Token'
      }
    });
    return response.data;
  }
  else if (modelName === '文心一言') {
    const response = await axios.post('文心一言API地址', {
      prompt: prompt
    }, {
      headers: {
        Authorization: 'Bearer 你的Token'
      }
    });
    return response.data;
  }
  // 更多模型按需添加
  else {
    throw new Error('不支持的模型');
  }
}

const PORT = 3000;
app.listen(PORT, () => {
  console.log(`后端服务器已启动，监听端口 ${PORT}`);
});
